package com.micromall.common.utils;

import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;

import java.io.*;
import java.util.Properties;
import java.util.UUID;

/**
 * ftp上传下载工具类
 * <p>Title: FtpUtil</p>
 * <p>Description: </p>
 * <p>Company: www.itcast.com</p>
 *
 * @version 1.0
 * @author 入云龙
 * @date 2015年7月29日下午8:11:51
 */
public class FtpUtil {

    /**
     * Description: 向FTP服务器上传文件
     *
     * @param host     FTP服务器hostname
     * @param port     FTP服务器端口
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param basePath FTP服务器基础目录
     * @param filePath FTP服务器文件存放路径。例如分日期存放：/2015/01/01。文件的路径为basePath+filePath
     * @param filename 上传到FTP服务器上的文件名
     * @param input    输入流
     * @return 成功返回true，否则返回false
     */
    public static boolean uploadFile(String host, int port, String username, String password, String basePath,
                                     String filePath, String filename, InputStream input) {
        boolean result = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(host, port);// 连接FTP服务器
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(username, password);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return result;
            }
            //切换到上传目录
            if (!ftp.changeWorkingDirectory(basePath + filePath)) {
                //如果目录不存在创建目录
                String[] dirs = filePath.split("/");
                String tempPath = basePath;
                for (String dir : dirs) {
                    if (null == dir || "".equals(dir)) continue;
                    tempPath += "/" + dir;
                    if (!ftp.changeWorkingDirectory(tempPath)) {
                        if (!ftp.makeDirectory(tempPath)) {
                            return result;
                        } else {
                            ftp.changeWorkingDirectory(tempPath);
                        }
                    }
                }
            }
            //设置上传文件的类型为二进制类型
            ftp.setFileType(FTP.BINARY_FILE_TYPE);
            //上传文件
            if (!ftp.storeFile(filename, input)) {
                return result;
            }
            input.close();
            ftp.logout();
            result = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }

    /**
     * Description: 从FTP服务器下载文件
     *
     * @param host       FTP服务器hostname
     * @param port       FTP服务器端口
     * @param username   FTP登录账号
     * @param password   FTP登录密码
     * @param remotePath FTP服务器上的相对路径
     * @param fileName   要下载的文件名
     * @param localPath  下载后保存到本地的路径
     * @return
     */
    public static boolean downloadFile(String host, int port, String username, String password, String remotePath,
                                       String fileName, String localPath) {
        boolean result = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(host, port);
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(username, password);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return result;
            }
            ftp.changeWorkingDirectory(remotePath);// 转移到FTP服务器目录
            FTPFile[] fs = ftp.listFiles();
            for (FTPFile ff : fs) {
                if (ff.getName().equals(fileName)) {
                    File localFile = new File(localPath + "/" + ff.getName());

                    OutputStream is = new FileOutputStream(localFile);
                    ftp.retrieveFile(ff.getName(), is);
                    is.close();
                }
            }

            ftp.logout();
            result = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }

    public static void domain(String[] args) {
        try {
//            FileInputStream in = new FileInputStream(new File("/tools/yirong.jpg"));
//            boolean flag = uploadFile("120.25.205.54", 21, "ftpuser", "admin", "/home/traffic/images", "/2017/04/24", "yirong.jpg", in);
//            System.out.println(flag);
//            InputStream inputstream = FtpUtil.class.getClassLoader().getResourceAsStream("properties/ftp.properties");
//            Properties properties = new Properties();
//            properties.load(inputstream);
//            String property = properties.getProperty("ftp.username");
//            System.out.println(property);
//              String value = PropertiesUtils.getValue("ftp.username");
//            System.out.println(value);
//            String propertiesValue = getPropertiesValue("properties/ftp.properties", "ftp.username");

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static String getPropertiesValue(String propertiesPath, String key) throws IOException {
        InputStream inputstream = FtpUtil.class.getClassLoader().getResourceAsStream(propertiesPath);
        Properties properties = new Properties();
        properties.load(inputstream);
        String value = properties.getProperty(key);
        return value;
    }

    public static void main(String[] args) {
        FTPClient ftpClient = new FTPClient();
        try {
            ftpClient.connect("120.25.205.54", 21);
            ftpClient.login("ftptraffic","admin");
            FileInputStream inputStream = new FileInputStream(new File(File.separator +"tools" + File.separator + "hahaha.jpg"));
            System.out.println(File.separator + "home" + File.separator + "ftpuser" + File.separator +
                    "traffic"+ File.separator + "images");
            ftpClient.changeWorkingDirectory(File.separator + "home" + File.separator + "traffic" + File.separator + "images" + File.separator);
            ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
            ftpClient.storeFile("yirong.jpg", inputStream);
            System.out.println("123");
            ftpClient.logout();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void down(String[] args) {
        FTPClient ftpClient = new FTPClient();
        FileOutputStream fos = null;

        try {
            ftpClient.connect("120.25.205.54");
            ftpClient.login("ftpuser", "21");

            String remoteFileName = "/home/ftpuser";
            // fos = new FileOutputStream("E:/test/test_back_081901.sql");
            fos = new FileOutputStream("/tools/ftp.jpg");
            ftpClient.setBufferSize(102400000);
            // 设置文件类型（二进制）
            ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE);
            ftpClient.retrieveFile(remoteFileName, fos);
        } catch (IOException e) {
            e.printStackTrace();
            throw new RuntimeException("FTP客户端出错！", e);
        } finally {
            IOUtils.closeQuietly(fos);
            try {
                ftpClient.disconnect();
            } catch (IOException e) {
                e.printStackTrace();
                throw new RuntimeException("关闭FTP连接发生异常！", e);
            }
        }
    }

    public static String getImgName(){
        return UUID.randomUUID().toString().replace("-","").substring(6) + ".jpg";
    }
}
